import os
from huggingface_hub import snapshot_download

# 目标路径：当前目录下的 all-MiniLM-L6-v2 文件夹
target_dir = os.path.join(os.getcwd(), "all-MiniLM-L6-v2")

print(f"正在下载模型到: {target_dir}")

snapshot_download(
    repo_id="sentence-transformers/all-MiniLM-L6-v2",
    local_dir=target_dir,
    endpoint="https://hf-mirror.com",  # 使用国内镜像
)

print("✅ 下载完成！")
print(f"模型路径: {target_dir}")